"""
需求：测特殊方法 on_start、 on_stop
"""

from locust import TaskSet, HttpLocust, task


# 1. 定义任务集和任务 ---- TaskSet、task
class Task(TaskSet):
    def index(self):
        print("启动，现在开始！！！")

    def index_end(self):
        print("结束！玩完了，也请会。")

    def on_start(self):
        self.index()

    def on_stop(self):
        self.index_end()
        
# 2. 定义用户行为  ----- HttpLocust
class User(HttpLocust):
    # 绑定任务集
    task_set = Task    # 千万不能加 ()

    # 设定用户与用户间 最小的延时时间, 单位：ms
    min_wait = 1000

    # 设定用户与用户间 最大的延时时间, 单位：ms
    max_wait = 3000

    # 指定主机域名
    host = "https://www.baidu.com"
